package com.jeecg.mineattribute.controller;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.io.RandomAccessFile;

import org.apache.commons.net.PrintCommandListener;
import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;



/**
 * 支持断点续传的FTP实用类
 * @author BenZhou
 * @version 0.1 实现基本断点上传下载
 * @version 0.2 实现上传下载进度汇报
 * @version 0.3 实现中文目录创建及中文文件创建，添加对于中文的支持
 */
public class ContinueFTP {

	
	
	public FTPClient ftpClient = new FTPClient();
	
	public ContinueFTP(){
		//设置将过程中使用到的命令输出到控制台
		this.ftpClient.addProtocolCommandListener(new PrintCommandListener(new PrintWriter(System.out)));
	}
	
	/**
	 * 连接到FTP服务器
	 * @param hostname 主机名
	 * @param port 端口
	 * @param username 用户名
	 * @param password 密码
	 * @return 是否连接成功
	 * @throws IOException
	 */
	public boolean connect(String hostname,int port,String username,String password) throws IOException{
		ftpClient.connect(hostname, port);
		ftpClient.setControlEncoding("GBK");
		if(FTPReply.isPositiveCompletion(ftpClient.getReplyCode())){
			if(ftpClient.login(username, password)){
				return true;
			}
		}
		disconnect();
		return false;
	}
	
	/**
	 * 上传文件到FTP服务器，支持断点续传
	 * @param local 本地文件名称，绝对路径
	 * @param remote 远程文件路径，使用/home/directory1/subdirectory/file.ext 按照Linux上的路径指定方式，支持多级目录嵌套，支持递归创建不存在的目录结构
	 * @return 上传结果
	 * @throws IOException
	 */
	public UploadStatus upload(String local,String remote) throws IOException{
		//设置PassiveMode传输
		ftpClient.enterLocalPassiveMode();
		//设置以二进制流的方式传输
		ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
		ftpClient.setControlEncoding("GBK");
		UploadStatus result;
		//对远程目录的处理
		String remoteFileName = remote;
		if(remote.contains("/")){
			remoteFileName = remote.substring(remote.lastIndexOf("/")+1);
			//创建服务器远程目录结构，创建失败直接返回
			if(CreateDirecroty(remote, ftpClient)==UploadStatus.Create_Directory_Fail){
				return UploadStatus.Create_Directory_Fail;
			}
		}
		
		//检查远程是否存在文件
		FTPFile[] files = ftpClient.listFiles(new String(remoteFileName.getBytes("GBK"),"iso-8859-1"));
		if(files.length == 1){
			long remoteSize = files[0].getSize();
			File f = new File(local);
			long localSize = f.length();
			if(remoteSize==localSize){
				return UploadStatus.File_Exits;
			}else if(remoteSize > localSize){
				return UploadStatus.Remote_Bigger_Local;
			}

			//尝试移动文件内读取指针,实现断点续传
			result = uploadFile(remoteFileName, f, ftpClient, remoteSize);
			
			//如果断点续传没有成功，则删除服务器上文件，重新上传
			if(result == UploadStatus.Upload_From_Break_Failed){
				if(!ftpClient.deleteFile(remoteFileName)){
					return UploadStatus.Delete_Remote_Faild;
				}
				result = uploadFile(remoteFileName, f, ftpClient, 0);
			}
		}else {
			result = uploadFile(remoteFileName, new File(local), ftpClient, 0);
		}
		return result;
	}
	/**
	 * 断开与远程服务器的连接
	 * @throws IOException
	 */
	public void disconnect() throws IOException{
		if(ftpClient.isConnected()){
			ftpClient.disconnect();
		}
	}
	
	/**
	 * 递归创建远程服务器目录
	 * @param remote 远程服务器文件绝对路径
	 * @param ftpClient FTPClient对象
	 * @return 目录创建是否成功
	 * @throws IOException
	 */
	public UploadStatus CreateDirecroty(String remote,FTPClient ftpClient) throws IOException{
		UploadStatus status = UploadStatus.Create_Directory_Success;
		String directory = remote.substring(0,remote.lastIndexOf("/")+1);
		if(!directory.equalsIgnoreCase("/")&&!ftpClient.changeWorkingDirectory(new String(directory.getBytes("GBK"),"iso-8859-1"))){
			//如果远程目录不存在，则递归创建远程服务器目录
			int start=0;
			int end = 0;
			if(directory.startsWith("/")){
				start = 1;
			}else{
				start = 0;
			}
			end = directory.indexOf("/",start);
			while(true){
				String subDirectory = new String(remote.substring(start,end).getBytes("GBK"),"iso-8859-1");
				if(!ftpClient.changeWorkingDirectory(subDirectory)){
					if(ftpClient.makeDirectory(subDirectory)){
						ftpClient.changeWorkingDirectory(subDirectory);
					}else {
						System.out.println("创建目录失败");
						return UploadStatus.Create_Directory_Fail;
					}
				}
				
				start = end + 1;
				end = directory.indexOf("/",start);
				
				//检查所有目录是否创建完毕
				if(end <= start){
					break;
				}
			}
		}
		return status;
	}
	
	/**
	 * 上传文件到服务器,新上传和断点续传
	 * @param remoteFile 远程文件名，在上传之前已经将服务器工作目录做了改变
	 * @param localFile 本地文件File句柄，绝对路径
	 * @param processStep 需要显示的处理进度步进值
	 * @param ftpClient FTPClient引用
	 * @return
	 * @throws IOException
	 */
	public UploadStatus uploadFile(String remoteFile,File localFile,FTPClient ftpClient,long remoteSize) throws IOException{
		UploadStatus status;
		//显示进度的上传
		long step = 1;
		if(localFile.length()>10*1024) 
		 step = localFile.length()/ 100;
		
		long process = 0;
		long localreadbytes = 0L;
		RandomAccessFile raf = new RandomAccessFile(localFile,"r");
		OutputStream out = ftpClient.appendFileStream(new String(remoteFile.getBytes("GBK"),"iso-8859-1"));
		//断点续传
		if(remoteSize>0){
			ftpClient.setRestartOffset(remoteSize);
			process = remoteSize /step;
			raf.seek(remoteSize);
			localreadbytes = remoteSize;
		}
		byte[] bytes = new byte[1024];
		int c;
		while((c = raf.read(bytes))!= -1){
			out.write(bytes,0,c);
			localreadbytes+=c;
			if(localreadbytes / step != process){
				process = localreadbytes / step;
				System.out.println("上传进度:" + process+"%");
				//TODO 汇报上传状态
			}
		}
		out.flush();
		raf.close();
		out.close();
		boolean result =ftpClient.completePendingCommand();
		if(remoteSize > 0){
			status = result?UploadStatus.Upload_From_Break_Success:UploadStatus.Upload_From_Break_Failed;
		}else {
			status = result?UploadStatus.Upload_New_File_Success:UploadStatus.Upload_New_File_Failed;
		}
		return status;
	}
	
//	public static void main(String[] args) {
//		ContinueFTP myFtp = new ContinueFTP();
//		try {
//			myFtp.connect("192.168.4.54", 21, "yh", "123456");
//			myFtp.ftpClient.makeDirectory(new String("文本".getBytes("GBK"),"iso-8859-1"));
//			myFtp.ftpClient.changeWorkingDirectory(new String("文本".getBytes("GBK"),"iso-8859-1"));
////			myFtp.ftpClient.makeDirectory(new String("爱你等于爱自己".getBytes("GBK"),"iso-8859-1"));
////			System.out.println(myFtp.upload("E:\\yw.flv", "/yw.flv",5));
//			System.out.println(myFtp.upload("f:\\bak\\tag1\\1.txt","/1.txt"));
//			myFtp.disconnect();
//		} catch (IOException e) {
//			System.out.println("连接FTP出错："+e.getMessage());
//		}
//	}


	/**
	 *  上传文件操作
	 *
	 * @param hostname ftp服务器ip
	 * @param port ftp端口
	 * @param username 登陆名
	 * @param password 密码
	 * @param directory ftp保存的文件夹名
	 * @param localPath 本地文件的绝对路径
	 * @param remotePath 远程路径,只需要文件名
	 */

	public void uploadFile(String hostname,int port,String username,String password,String directory,String localPath,String remotePath){
		try {
			connect(hostname, port, username, password);
			ftpClient.makeDirectory(new String(directory.getBytes("GBK"),"iso-8859-1"));
			ftpClient.changeWorkingDirectory(new String(directory.getBytes("GBK"),"iso-8859-1"));
			//上传文件前删掉存在的文件
			ftpClient.deleteFile(remotePath);
			
			System.out.println(upload(localPath,remotePath));
			disconnect();
		} catch (IOException e) {
			System.out.println("连接FTP出错："+e.getMessage());
		}
	}
}